package com.baidu.ueditor;

import com.apache.tools.StrUtil;
import com.apache.uct.common.ToolsUtil;
import com.baidu.ueditor.define.ActionMap;
import org.apache.log4j.Logger;
import org.json.JSONArray;
import org.json.JSONObject;

import java.io.*;
import java.util.HashMap;
import java.util.Map;

/**
 * 配置管理器
 * @author hancong03@baidu.com
 *
 */
public final class ConfigManager {
	private Logger log = Logger.getLogger(getClass());

	private final String rootPath;
	private final String originalPath;
	private final String contextPath;
	private static final String configFileName = "config.json";
	private String parentPath = null;
	private JSONObject jsonConfig = null;
	// 涂鸦上传filename定义
	private final static String SCRAWL_FILE_NAME = "scrawl";
	// 远程图片抓取filename定义
	private final static String REMOTE_FILE_NAME = "remote";

	/*
	 * 通过一个给定的路径构建一个配置管理器， 该管理器要求地址路径所在目录下必须存在config.properties文件
	 */
	private ConfigManager(String rootPath, String contextPath, String uri,Map<String,String> paramMap) throws FileNotFoundException, IOException {

		rootPath = rootPath.replace("\\", "/");

		this.rootPath = rootPath;
		this.contextPath = contextPath;

		if (contextPath.length() > 0) {
			this.originalPath = this.rootPath + uri.substring(contextPath.length());
		} else {
			this.originalPath = this.rootPath + uri;
		}

		this.initEnv(paramMap);

	}

	/**
	 * 配置管理器构造工厂
	 * @param rootPath 服务器根路径
	 * @param contextPath 服务器所在项目路径
	 * @param uri 当前访问的uri
	 * @return 配置管理器实例或者null
	 */
	public static ConfigManager getInstance(String rootPath, String contextPath, String uri, Map<String,String> paramMap) {

		try {
			return new ConfigManager(rootPath, contextPath, uri,paramMap);
		} catch (Exception e) {
			e.printStackTrace();
			return null;
		}

	}

	// 验证配置文件加载是否正确
	public boolean valid() {
		return this.jsonConfig != null;
	}

	public JSONObject getAllConfig() {

		return this.jsonConfig;

	}

	public Map<String, Object> getConfig(int type) throws Exception {

		Map<String, Object> conf = new HashMap<String, Object>();
		String savePath = null;

		switch (type) {

		case ActionMap.UPLOAD_FILE:
			conf.put("isBase64", "false");
			conf.put("maxSize", this.jsonConfig.getLong("fileMaxSize"));
			conf.put("allowFiles", this.getArray("fileAllowFiles"));
			conf.put("fieldName", this.jsonConfig.getString("fileFieldName"));
			savePath = this.jsonConfig.getString("filePathFormat");
			break;

		case ActionMap.UPLOAD_IMAGE:
			conf.put("isBase64", "false");
			conf.put("maxSize", this.jsonConfig.getLong("imageMaxSize"));
			conf.put("allowFiles", this.getArray("imageAllowFiles"));
			conf.put("fieldName", this.jsonConfig.getString("imageFieldName"));
			savePath = this.jsonConfig.getString("imagePathFormat");
			break;

		case ActionMap.UPLOAD_VIDEO:
			conf.put("maxSize", this.jsonConfig.getLong("videoMaxSize"));
			conf.put("allowFiles", this.getArray("videoAllowFiles"));
			conf.put("fieldName", this.jsonConfig.getString("videoFieldName"));
			savePath = this.jsonConfig.getString("videoPathFormat");
			break;

		case ActionMap.UPLOAD_SCRAWL:
			conf.put("filename", ConfigManager.SCRAWL_FILE_NAME);
			conf.put("maxSize", this.jsonConfig.getLong("scrawlMaxSize"));
			conf.put("fieldName", this.jsonConfig.getString("scrawlFieldName"));
			conf.put("isBase64", "true");
			savePath = this.jsonConfig.getString("scrawlPathFormat");
			break;

		case ActionMap.CATCH_IMAGE:
			conf.put("filename", ConfigManager.REMOTE_FILE_NAME);
			conf.put("filter", this.getArray("catcherLocalDomain"));
			conf.put("maxSize", this.jsonConfig.getLong("catcherMaxSize"));
			conf.put("allowFiles", this.getArray("catcherAllowFiles"));
			conf.put("fieldName", this.jsonConfig.getString("catcherFieldName") + "[]");
			savePath = this.jsonConfig.getString("catcherPathFormat");
			break;

		case ActionMap.LIST_IMAGE:
			conf.put("allowFiles", this.getArray("imageManagerAllowFiles"));
			conf.put("dir", this.jsonConfig.getString("imageManagerListPath"));
			conf.put("count", this.jsonConfig.getInt("imageManagerListSize"));
			break;

		case ActionMap.LIST_FILE:
			conf.put("allowFiles", this.getArray("fileManagerAllowFiles"));
			conf.put("dir", this.jsonConfig.getString("fileManagerListPath"));
			conf.put("count", this.jsonConfig.getInt("fileManagerListSize"));
			break;

		}

		conf.put("savePath", savePath);
		//conf.put( "rootPath", this.rootPath );
		conf.put("rootPath", this.jsonConfig.getString("imageUploadPrefix"));
		if (this.jsonConfig.getString("imageUploadPrefix") == null
				|| this.jsonConfig.getString("imageUploadPrefix").equals("")) {
			conf.put("rootPath", this.rootPath);
		}
		return conf;

	}

	private void initEnv(Map<String,String> paramMap) throws FileNotFoundException, IOException {

		File file = new File(this.originalPath);

		if (!file.isAbsolute()) {
			file = new File(file.getAbsolutePath());
		}

		this.parentPath = file.getParent();

		String configContent = this.readFile(this.getConfigPath());
		try {
			JSONObject jsonConfig = new JSONObject(configContent);
			log.info("ueditor 接收前端特殊参数="+paramMap);
			String file_directory = StrUtil.doNull(paramMap.get("file_directory"),"default");
			/**
			 * add 20180428 ,为了每次不更改或者部署生产环境忘记更改config.json中相关配置
			 * 在此重新根据config.properties配置文件替换json相关参数。
			 * */
			String uploadPrefix = ToolsUtil.getInstance().getLocalByKey("config.properties","UPLOAD_DIRECTORY");
			String downloadPrefix = ToolsUtil.getInstance().getLocalByKey("config.properties","DOWNLOAD_URL");
			String new_uploadPrefix = StrUtil.doNull(uploadPrefix,"E:/upload")+"/atta/"+file_directory;
			String new_downloadPrefix = StrUtil.doNull(downloadPrefix,"http://127.0.0.1:11080")+"/atta/"+file_directory;
			jsonConfig.put("imageUploadPrefix",new_uploadPrefix);
			jsonConfig.put("imageUrlPrefix",new_downloadPrefix);
			jsonConfig.put("videoUrlPrefix",new_downloadPrefix);
			jsonConfig.put("fileUrlPrefix",new_downloadPrefix);
			//后续可以在此增加其他类型的目录，地址等（参照config.json）
			log.info("ueditor imageUploadPrefix="+jsonConfig.get("imageUploadPrefix"));
			log.info("ueditor imageUrlPrefix="+jsonConfig.get("imageUrlPrefix"));
			log.info("ueditor fileUrlPrefix="+jsonConfig.get("fileUrlPrefix"));
			this.jsonConfig = jsonConfig;
		} catch (Exception e) {
			this.jsonConfig = null;
		}

	}

	private String getConfigPath() {
		String configPath = "";
		configPath = this.parentPath + File.separator + ConfigManager.configFileName;
		boolean isEx = new File(configPath).exists();
		log.info("configPath-->"+configPath);
		if (!isEx){
			configPath = ToolsUtil.getClassLoaderPath()+ "config/ueditor-config.json";
			log.info("因以上文件不存在，则查找如下目录文件："+configPath);
		}
		return configPath;
	}

	private String[] getArray(String key) throws Exception {

		JSONArray jsonArray = this.jsonConfig.getJSONArray(key);
		String[] result = new String[jsonArray.length()];

		for (int i = 0, len = jsonArray.length(); i < len; i++) {
			result[i] = jsonArray.getString(i);
		}

		return result;

	}

	private String readFile(String path) throws IOException {

		StringBuilder builder = new StringBuilder();

		try {

			InputStreamReader reader = new InputStreamReader(new FileInputStream(path), "UTF-8");
			BufferedReader bfReader = new BufferedReader(reader);

			String tmpContent = null;

			while ((tmpContent = bfReader.readLine()) != null) {
				builder.append(tmpContent);
			}

			bfReader.close();

		} catch (UnsupportedEncodingException e) {
			// 忽略
		}

		return this.filter(builder.toString());

	}

	// 过滤输入字符串, 剔除多行注释以及替换掉反斜杠
	private String filter(String input) {

		return input.replaceAll("/\\*[\\s\\S]*?\\*/", "");

	}

}
